import pandas as pd
import matplotlib.pyplot as plt

# 设置中文字体
plt.rcParams['font.family'] = 'SimHei'

# 读取Excel文件
file_path = '../雪龙集团资产负债表.xlsx'
data = pd.ExcelFile(file_path)

# 加载数据并清理列名
df = data.parse('Sheet1')
df.columns = ["项目"] + [2020, 2021, 2022, 2023]

# 转换数据为数值格式，去除逗号和非数值字符
for col in [2020, 2021, 2022, 2023]:
    df[col] = pd.to_numeric(df[col].str.replace(',', '', regex=True), errors='coerce')

# 提取需要绘图的指标
metrics_of_interest = ['应收帐款(元)', '资产总计(元)', '负债合计(元)']
data_to_plot = df[df["项目"].isin(metrics_of_interest)]

# 绘制折线图
plt.figure(figsize=(10, 6))
for index, row in data_to_plot.iterrows():
    plt.plot([2020, 2021, 2022, 2023], row[1:], marker='o', label=row["项目"])

# 添加图表标题和标签
plt.title("雪龙集团2020-2023年应收账款_资产总计_负债合计折线图", fontsize=16)
plt.xlabel("年份", fontsize=12)
plt.ylabel("金额（百万元）", fontsize=12)

# 设置y轴为科学记数法显示，便于阅读大数
plt.ticklabel_format(style='sci', axis='y', scilimits=(6, 6))

# 添加图例和网格
plt.legend(loc="best", fontsize=10)
plt.grid(alpha=0.5)
plt.tight_layout()

# 保存图表
plt.savefig('../分析图象/应收账款_资产总计_负债合计图像.png')

# 显示图表
plt.show()
